package com.hspedu.set_;

import java.util.Comparator;
import java.util.TreeSet;

/**
 * @Author Stringzhua
 * @Date 2024/3/13 17:03
 * description:
 */
@SuppressWarnings("all")
public class TreeSet_ {
    public static void main(String[] args) {
        //1.当使用无参构造器创建TreeSet时,仍然是无序的
        //2.希望添加的元素,按照字符串大小来排序
        //3.使用TreeSet提供的构造器,可以传入一个比较器（匿名内部类）
        //并指定排序位置
        //4.查看源码
        //按字母表排序从z->a
//        TreeSet treeSet = new TreeSet();
        TreeSet treeSet=new TreeSet(new Comparator() {//匿名内部类
            @Override
            public int compare(Object o1, Object o2) {
                //下面 调用String的compareTo来进行字符串大小的比较
                return ((String)o2).compareTo((String)o1);
            }
        });
        //添加数据
        treeSet.add("jack");
        treeSet.add("tom");
        treeSet.add("sp");
        treeSet.add("a");
        treeSet.add("tom");//替换

        System.out.println("treeSet = " + treeSet);//无顺序

        /*
        * 源码解读:
        * 1.构造器把传入的比较器对象,赋给了TreeSet的底层 TreeMap的属性
        *  public TreeMap(Comparator<? super K> comparator) {
                this.comparator = comparator;
            }
        * 2.在调用treeSet.add("tom")方法时,在底层会执行到
        *  if (cpr != null) { //cpr 就是我们的匿名内部类
                do {
                    parent = t;
                    cmp = cpr.compare(key, t.key);//动态绑定到我们的匿名内部类的compare方法
                    if (cmp < 0)
                        t = t.left;
                    else if (cmp > 0)
                        t = t.right;
                    else //如果相等,即返回0,这个Key就没有加入
                        return t.setValue(value);
                } while (t != null);
            }
        *
        * */

        //如果要求加入的元素,按长度大小排序 abc加不进去,添加失败,在源码中直接被返回
        TreeSet treeSet1=new TreeSet(new Comparator() {
            @Override
            public int compare(Object o1, Object o2) {
//                return ((String)o2).length()-((String)o1).length();//[从大到小排序]
                return ((String)o1).length()-((String)o2).length();//[从小到大排序]
            }
        });

        treeSet1.add("jack");
        treeSet1.add("tom");//3
        treeSet1.add("sp");
        treeSet1.add("a");
        treeSet1.add("abc");//3 abc加不进去,添加失败,在源码中直接被返回

        System.out.println("treeSet1 = " + treeSet1);

    }
}
